package com.yxinsur.imc.common.utils.file;

import cn.hutool.core.util.RandomUtil;
import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClient;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.Bucket;
import com.aliyun.oss.model.OSSObject;
import com.aliyun.oss.model.ObjectMetadata;
import com.aliyun.oss.model.PutObjectResult;
import com.yxinsur.imc.common.utils.DateUtils;
import com.yxinsur.imc.common.utils.SecurityUtils;
import io.minio.MinioClient;
import io.minio.PutObjectArgs;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.multipart.MultipartFile;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.net.URLDecoder;
import java.nio.charset.StandardCharsets;

/**
 * 使用阿里云OSS存储对象上传文件等工具类
 */

@Configuration
//@PropertySource("classpath:config/oss.properties")
public class OSSClientUtil {

    /**
     * 日志
     */
    private static Logger logger = LoggerFactory.getLogger(OSSClientUtil.class);

    /**
     * 阿里云API的密钥Access Key ID
     */
    @Value("${oss.accessKeyId}")
    private static String accessKeyId;
    /**
     * 阿里云API的密钥Access Key Secret
     */
    @Value("${oss.accessKeySecret}")
    private static String accessKeySecret;
    /**
     * 阿里云API的内或外网域名
     */
    @Value("${oss.endpoint}")
    private static String endpoint;
    /**
     * 阿里云API的bucket名称
     */
    @Value("${oss.bucketName}")
    private static String bucketName;
    /**
     * 阿里云API的文件夹名称
     */
    @Value("${oss.folder}")
    private static String folder;

    @Value("${oss.type}")
    private static String type;

    @Value("${oss.ossUrl}")
    private static String ossUrl;

//    public static MinioClient minioClient;

    /**
     * 获取阿里云OSS客户端对象
     *
     * @return ossClient
     */
    public static OSSClient getOSSClient() {

        return new OSSClient(endpoint, accessKeyId, accessKeySecret);
    }

    /**
     * 创建存储空间
     *
     * @param ossClient  OSS连接
     * @param bucketName 存储空间
     * @return
     */
    public static String createBucketName(OSSClient ossClient, String bucketName) {
        // 存储空间
        final String bucketNames = bucketName;
        if (!ossClient.doesBucketExist(bucketName)) {
            // 创建存储空间
            Bucket bucket = ossClient.createBucket(bucketName);
            logger.info("创建存储空间成功");
            return bucket.getName();
        }
        return bucketNames;
    }

    /**
     * 删除存储空间buckName
     *
     * @param ossClient  oss对象
     * @param bucketName 存储空间
     */
    public static void deleteBucket(OSSClient ossClient, String bucketName) {
        ossClient.deleteBucket(bucketName);
        logger.info("删除" + bucketName + "Bucket成功");
    }

    /**
     * 创建模拟文件夹
     *
     * @param ossClient  oss连接
     * @param bucketName 存储空间
     * @param folder     模拟文件夹名如"qj_nanjing/"
     * @return 文件夹名
     */
    public static String createFolder(OSSClient ossClient, String bucketName, String folder) {
        // 文件夹名
        final String keySuffixWithSlash = folder;
        // 判断文件夹是否存在，不存在则创建
        if (!ossClient.doesObjectExist(bucketName, keySuffixWithSlash)) {
            // 创建文件夹
            ossClient.putObject(bucketName, keySuffixWithSlash, new ByteArrayInputStream(new byte[0]));
            logger.info("创建文件夹成功");
            // 得到文件夹名
            OSSObject object = ossClient.getObject(bucketName, keySuffixWithSlash);
            String fileDir = object.getKey();
            return fileDir;
        }
        return keySuffixWithSlash;
    }

    /**
     * 根据key删除OSS服务器上的文件
     *
     * @param ossClient  oss连接
     * @param bucketName 存储空间
     * @param folder     模拟文件夹名 如"qj_nanjing/"
     * @param key        Bucket下的文件的路径名+文件名 如："upload/cake.jpg"
     */
    public static void deleteFile(OSSClient ossClient, String bucketName, String folder, String key) {
        ossClient.deleteObject(bucketName, folder + key);
        logger.info("删除" + bucketName + "下的文件" + folder + key + "成功");
    }

    /**
     * 上传图片至OSS
     *
     * @param ossClient  oss连接
     * @param file       上传文件（文件全路径如：D:\\image\\cake.jpg）
     * @return String 返回的唯一MD5数字签名
     */
    public static String uploadObject2OSS(OSSClient ossClient, File file) {
        String resultStr = null;
        try {
            // 以输入流的形式上传文件
            InputStream is = new FileInputStream(file);
            // 文件名
            String fileName = file.getName();
            // 文件大小
            Long fileSize = file.length();
            // 创建上传Object的Metadata
            ObjectMetadata metadata = new ObjectMetadata();
            // 上传的文件的长度
            metadata.setContentLength(is.available());
            // 指定该Object被下载时的网页的缓存行为
            metadata.setCacheControl("no-cache");
            // 指定该Object下设置Header
            metadata.setHeader("Pragma", "no-cache");
            // 指定该Object被下载时的内容编码格式
            metadata.setContentEncoding("utf-8");
            // 文件的MIME，定义文件的类型及网页编码，决定浏览器将以什么形式、什么编码读取文件。如果用户没有指定则根据Key或文件名的扩展名生成，
            // 如果没有扩展名则填默认值application/octet-stream
            metadata.setContentType(getContentType(fileName));
            // 指定该Object被下载时的名称（指示MINME用户代理如何显示附加的文件，打开或下载，及文件名称）
            metadata.setContentDisposition("filename/filesize=" + fileName + "/" + fileSize + "Byte.");
            // 上传文件 (上传文件流的形式)
            PutObjectResult putResult = ossClient.putObject(bucketName, folder + fileName, is, metadata);
            // 解析结果
            resultStr = putResult.getETag();
        } catch (Exception e) {
            e.printStackTrace();
            logger.error("上传阿里云OSS服务器异常." + e.getMessage(), e);
        }
        return resultStr;
    }
    /**
     * 上传xml流至OSS
     *
     * @param inputStream
     * @return String 返回的唯一MD5数字签名
     */
    public static String uploadExcel(InputStream inputStream,String typeName) {
        String resultStr = null;
        OSS ossClient = null;
        MinioClient minioClient = null;
        try {
            if("aliyun".equals(type)){
                // 创建OSS实例
                ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
            }
            if("minio".equals(type)){
                // 创建minio实例
                minioClient = MinioClient.builder().endpoint(endpoint).credentials(accessKeyId, accessKeySecret).build();
            }
            String dateTime = typeName+"-"+ SecurityUtils.getTenantId()+"-"+DateUtils.parseDateToStr(DateUtils.YYYYMMDDHHMMSS,DateUtils.getNowDate());
            dateTime = dateTime + RandomUtil.randomChar() + RandomUtil.randomNumbers(3)+".xlsx";
            // 填写Bucket名称和Object完整路径。Object完整路径中不能包含Bucket名称。
            if("aliyun".equals(type)){
                ossClient.putObject(bucketName,dateTime,inputStream);
            }
            if("minio".equals(type)){
                minioClient.putObject(PutObjectArgs.builder().bucket(bucketName).object(dateTime).stream(inputStream, inputStream.available(), -1).contentType("application/octet-stream").build());
            }
            resultStr = dateTime;
        } catch (Exception e) {
            e.printStackTrace();
            logger.error("上传阿里云OSS服务器异常." + e.getMessage(), e);
        } finally {
            // 关闭OSSClient数据源
            if("aliyun".equals(type))ossClient.shutdown();
        }
        return resultStr;
    }

    /**
     * 上传文件至OSS
     *
     * @param file 文件
     * @return String 返回的唯一MD5数字签名
     */
    public static String upload2OSS(MultipartFile file) {
        OSS ossClient = null;
        MinioClient minioClient = null;
        try {
            if("aliyun".equals(type)){
                // 创建OSS实例
                ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
            }
            if("minio".equals(type)){
                // 创建minio实例
                minioClient = MinioClient.builder().endpoint(endpoint).credentials(accessKeyId, accessKeySecret).build();
            }
            // 获取上传文件输入流
            InputStream inputStream = file.getInputStream();//file.getInputStream();
            // 获取文件名称
            String fileName = FileUploadUtils.extractFilename(file);
            String endWith = fileName.substring(fileName.lastIndexOf(".")+1);
            // 在文件名称里面添加随机唯一的值,避免文件覆盖
            // 获取当前日期,给日期格式化，可以使文件以2021/3/11分类
            String dateTime = DateUtils.parseDateToStr(DateUtils.YYYYMMDDHHMMSS,DateUtils.getNowDate());
            dateTime = dateTime + RandomUtil.randomChar() + RandomUtil.randomNumbers(3);
            // 拼接
            fileName=dateTime+"."+endWith;
            // 调用oss方法实现上传
            // 第一个参数  Bucket名称
            // 第二个参数  上传到oss文件路径和文件名称   aa/bb/1.jpg
            // 第三个参数  上传文件输入流
            if("aliyun".equals(type)){
                ossClient.putObject(bucketName,fileName,inputStream);
            }
            if("minio".equals(type)){
                minioClient.putObject(PutObjectArgs.builder().bucket(bucketName).object(fileName).stream(inputStream, inputStream.available(), -1).contentType(file.getContentType()).build());
            }
            // 把上传后的路径返回存到数据库
            return ossUrl+"/"+fileName;
        }catch (Exception e){
            e.printStackTrace();
        }finally {
            // 关闭OSSClient数据源
            if("aliyun".equals(type))ossClient.shutdown();
        }
        return null;
    }

    /**
     * 通过文件名判断并获取OSS服务文件上传时文件的contentType
     *
     * @param fileName 文件名
     * @return 文件的contentType
     */
    public static String getContentType(String fileName) {
        // 文件的后缀名
        String fileExtension = fileName.substring(fileName.lastIndexOf("."));
        if (".bmp".equalsIgnoreCase(fileExtension)) {
            return "image/bmp";
        }
        if (".gif".equalsIgnoreCase(fileExtension)) {
            return "image/gif";
        }
        if (".jpeg".equalsIgnoreCase(fileExtension) || ".jpg".equalsIgnoreCase(fileExtension)
                || ".png".equalsIgnoreCase(fileExtension)) {
            return "image/jpeg";
        }
        if (".png".equalsIgnoreCase(fileExtension)) {
            return "image/png";
        }
        if (".html".equalsIgnoreCase(fileExtension)) {
            return "text/html";
        }
        if (".txt".equalsIgnoreCase(fileExtension)) {
            return "text/plain";
        }
        if (".vsd".equalsIgnoreCase(fileExtension)) {
            return "application/vnd.visio";
        }
        if (".ppt".equalsIgnoreCase(fileExtension) || "pptx".equalsIgnoreCase(fileExtension)) {
            return "application/vnd.ms-powerpoint";
        }
        if (".doc".equalsIgnoreCase(fileExtension) || "docx".equalsIgnoreCase(fileExtension)) {
            return "application/msword";
        }
        if (".xml".equalsIgnoreCase(fileExtension)) {
            return "text/xml";
        }
        // 默认返回类型
        return "";
    }
    //获取地址前缀
    public static String getPrePath() {
        return ossUrl;
    }

    @Value("${oss.accessKeyId}")
    public void setAccessKeyId(String accessKeyId) {
        OSSClientUtil.accessKeyId = accessKeyId;
    }

    @Value("${oss.accessKeySecret}")
    public void setAccessKeySecret(String accessKeySecret) {
        OSSClientUtil.accessKeySecret = accessKeySecret;
    }

    @Value("${oss.endpoint}")
    public void setEndpoint(String endpoint) {
        OSSClientUtil.endpoint = endpoint;
    }

    @Value("${oss.bucketName}")
    public void setBucketName(String bucketName) {
        OSSClientUtil.bucketName = bucketName;
    }

    @Value("${oss.folder}")
    public void setFolder(String folder) {
        OSSClientUtil.folder = folder;
    }

    @Value("${oss.type}")
    public void setType(String type) {
        OSSClientUtil.type = type;
    }

    @Value("${oss.ossUrl}")
    public void setOssUrl(String ossUrl) {
        OSSClientUtil.ossUrl = ossUrl;
    }
}